package com.module.ble.repo

import com.module.ble.db.BloodOxygenTableQueryHelper
import com.module.ble.db.HealthExaManager
import com.module.ble.db.table.UserBloodOxygen
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.MutableStateFlow
import kotlinx.coroutines.flow.flatMapLatest
import kotlinx.coroutines.flow.flowOf
import kotlinx.coroutines.launch

/**
 * 作者：sosou
 * <p>
 * 版本：1.0
 * <p>
 * 创建日期：2024/10/18
 * <p>
 * 描述：仓库类，所有心率相关的数据操作都在此类中进行，数据库的增、删、改、查操作
 * <p>
 * 修订历史：
 */
class BloodOxygenRepo private constructor() {

    private val refreshTrigger = MutableStateFlow(0)

    companion object {
        val instance by lazy(mode = LazyThreadSafetyMode.SYNCHRONIZED) { BloodOxygenRepo() }
    }

    private val dao = HealthExaManager.mBloodOxygenDao

    @OptIn(ExperimentalCoroutinesApi::class)
    private val scope = CoroutineScope(Dispatchers.IO.limitedParallelism(1))

    /**
     * 查找所有数据
     */
    fun findAll(): MutableList<UserBloodOxygen>? {
        //数据库查找所有数据
        return dao?.findAll()
    }

    /**
     * 查找所有未上传数据
     */
    fun findAllNotUp(): MutableList<UserBloodOxygen>? {
        //数据库查找所有数据
        return dao?.findAllNotUp()
    }

    /**
     * 查找所有数据
     */
    @OptIn(ExperimentalCoroutinesApi::class)
    fun findAllFlow(): Flow<List<UserBloodOxygen>> = refreshTrigger.flatMapLatest {
        dao?.findLatestDayFlow() ?: flowOf(emptyList())
    }

    /**
     * 根据时间范围查询数据
     */
    fun findByTimeRange(startTime:Long, endTime:Long): List<UserBloodOxygen>? {
        //数据库查找所有数据
        return dao?.findByTimeRange(startTime,endTime)
    }

    /**
     * 根据时间范围查询数据（Flow版本）
     */
    @OptIn(ExperimentalCoroutinesApi::class)
    fun findByTimeRangeFlow(startTime:Long, endTime:Long): Flow<List<UserBloodOxygen>> = refreshTrigger.flatMapLatest { dao?.findByTimeRangeFlow(startTime,endTime) ?: flowOf(emptyList())}

    /**
     * 根据多个时间段查询睡眠数据
     * @param timeRanges 时间段数组，每个Pair的first为开始时间戳，second为结束时间戳
     * @return 查询到的睡眠数据列表
     */
    fun findByMultiTimeRanges(timeRanges: List<Pair<Long, Long>>): List<UserBloodOxygen>? {
        if (timeRanges.isEmpty()) return emptyList()

        val query = BloodOxygenTableQueryHelper.createMultiTimeRangeQuery(timeRanges)
        return dao?.findByCustomQuery(query)
    }

    /**
     * 根据多个时间段查询睡眠数据（Flow版本）
     * @param timeRanges 时间段数组，每个Pair的first为开始时间戳，second为结束时间戳
     * @return Flow<List<UserBloodOxygen>>
     */
    @OptIn(ExperimentalCoroutinesApi::class)
    fun findByMultiTimeRangesFlow(timeRanges: List<Pair<Long, Long>>): Flow<List<UserBloodOxygen>> {
        if (timeRanges.isEmpty()) return flowOf(emptyList())

        return refreshTrigger.flatMapLatest {
            val query = BloodOxygenTableQueryHelper.createMultiTimeRangeQuery(timeRanges)
            dao?.findByCustomQueryFlow(query) ?: flowOf(emptyList())
        }
    }

    /**
     * 根据传入时间戳查询该时间戳当天的所有数据
     */
    fun findByOneDay(timestamp: Long): List<UserBloodOxygen>? {
        //数据库查找所有数据
        return dao?.findByDay(timestamp)
    }

    /**
     * 根据传入时间戳查询该时间戳当天的最新数据
     */
    fun findLatestByDay(timestamp: Long): UserBloodOxygen? {
        //数据库查找所有数据
        return dao?.findLatestByDay(timestamp)
    }

    /**
     * 插入数据
     */
    fun add2DB(data: UserBloodOxygen?) {
        data?.let {
            scope.launch {
                dao?.insert(it)
            }
        }
    }


    /**
     * 删除数据库中数据
     */
    private fun delAll2DB() {
        scope.launch {
            dao?.deleteAll()
        }
    }

    fun insertDistinct(mList: List<UserBloodOxygen>, batchSize: Int = 500) {
        if (mList.isNullOrEmpty()) return
        scope.launch {
            // 分批处理
            mList.chunked(batchSize).forEach { batch ->
                dao?.insert(batch) // 直接插入整个批次
            }
        }
    }

    fun updateBatch(mList: List<UserBloodOxygen>, batchSize: Int = 500) {
        if (mList.isNullOrEmpty()) return
        scope.launch {
            // 分批处理
            mList.chunked(batchSize).forEach { batch ->
                dao?.update(batch) // 分配更新
            }
        }
    }

    /**
     * 自动触发 Flow 的更新，进而更新 LiveData
     */
    fun refreshData() {
        refreshTrigger.value += 1  // 增加触发器的值，这会导致 Flow 重新发射
    }

    /**
     *
     */
    fun updateAllFirmNameToOne() {
        scope.launch {
            dao?.updateAllFirmNameToOne()
        }
    }

    /**
     *
     */
    fun updateAllSysStatToZero() {
        scope.launch {
            dao?.updateAllSysStatToZero()
        }
    }

    /**
     *
     */
    fun deleteInvalidBloodOxygenData() {
        scope.launch {
            dao?.deleteInvalidBloodOxygenData()
        }
    }
}